The returning Diego Costa played a key role as Chelsea earned a much-needed win over struggling Aston Villa.