diff --git a/src/2024/day2/rust/mod.rs b/src/2024/day2/rust/mod.rs
index 1f81fe8..703ab45 100644
--- a/src/2024/day2/rust/mod.rs
+++ b/src/2024/day2/rust/mod.rs
@@ -1,98 +1,81 @@
-pub fn validate_report(report: &Vec<i32>) -> bool {
-    let mut valid = true;
-    let direction = {
-        if report[1] > report[0] {
-            1
-        } else {
-            -1
-        }
-    };
-    let mut diff = 0;
+pub fn validate_report(report: &[i32]) -> bool {
+    let direction = if report[1] > report[0] { 1 } else { -1 };
     for i in 1..report.len() {
-        if report[i] > report[i - 1] {
-            if direction != 1 {
-                valid = false;
-                break;
-            }
-        } else {
-            if direction != -1 {
-                valid = false;
-                break;
-            }
-        }
-        diff = (report[i] - report[i - 1]).abs();
-        if diff < 1 || diff > 3 {
-            valid = false;
-            break;
+        let diff = report[i] - report[i - 1];
+        if diff.abs() < 1
+            || diff.abs() > 3
+            || (direction == 1 && diff < 0)
+            || (direction == -1 && diff > 0)
+        {
+            return false;
         }
     }
-    if valid {
-        return true;
-    } else {
-        return false;
-    }
+    true
 }
 
-pub fn part_a(input: &str) -> i32 {
-    let reports: Vec<Vec<i32>> = input
+fn parse_reports(input: &str) -> Vec<Vec<i32>> {
+    input
         .lines()
         .map(|line| {
             line.split_whitespace()
-                .map(|num| num.parse().unwrap())
+                .map(|num| num.parse().expect("should be a number"))
                 .collect()
         })
-        .collect();
-    let mut safe_reports = 0;
-    for report in reports.iter() {
-        if validate_report(report) {
-            safe_reports += 1;
-        }
-    }
-    safe_reports
+        .collect()
+}
+
+pub fn part_a(input: &str) -> i32 {
+    parse_reports(input)
+        .iter()
+        .filter(|report| validate_report(report))
+        .count() as i32
 }
 
 pub fn part_b(input: &str) -> i32 {
-    let reports: Vec<Vec<i32>> = input
-        .lines()
-        .map(|line| {
-            line.split_whitespace()
-                .map(|num| num.parse().unwrap())
-                .collect()
-        })
-        .collect();
-    let mut safe_reports = 0;
-    for report in reports.iter() {
-        if validate_report(report) {
-            safe_reports += 1;
-        } else {
-            let mut new_reports: Vec<Vec<i32>> = Vec::new();
-            for i in 0..report.len() {
-                let mut new_report = report.clone();
+    parse_reports(input)
+        .iter()
+        .filter(|report| {
+            (0..report.len()).any(|i| {
+                let mut new_report = report.to_vec();
                 new_report.remove(i);
-                new_reports.push(new_report);
-            }
-            for new_report in new_reports.iter() {
-                if validate_report(new_report) {
-                    safe_reports += 1;
-                    break;
-                }
-            }
-        }
-    }
-    safe_reports
+                validate_report(&new_report)
+            })
+        })
+        .count() as i32
 }
 
 pub fn main() {
     let input = include_str!("../input.txt");
     let example_input = include_str!("../input_example.txt");
     // let example_input_b = include_str!("../input_example_b.txt");
-    let day: u8 = 1;
+    let day: u8 = 2;
     println!("Day {} Example Part A: {}", day, part_a(example_input));
     println!("Day {} Part A: {}", day, part_a(input));
     println!("Day {} Example Part B: {}", day, part_b(example_input));
     println!("Day {} Part B: {}", day, part_b(input));
 }
 
-// #[cfg(test)]
-// mod tests {
-// }
+#[cfg(test)]
+mod tests {
+    use super::*;
+    #[test]
+    fn test_part_a_example() {
+        let example_input = include_str!("../input_example.txt");
+        assert_eq!(part_a(example_input), 2);
+    }
+    #[test]
+    fn test_part_a() {
+        let input = include_str!("../input.txt");
+        assert_eq!(part_a(input), 502);
+    }
+    #[test]
+    fn test_part_b_example() {
+        let example_input_b = include_str!("../input_example.txt");
+        assert_eq!(part_b(example_input_b), 4);
+    }
+    #[test]
+    fn test_part_b() {
+        let input = include_str!("../input.txt");
+        assert_eq!(part_b(input), 544);
+    }
+}