diff --git a/src/bin/01.rs b/src/bin/01.rs
index 2c13fe5..9a0f9d0 100644
--- a/src/bin/01.rs
+++ b/src/bin/01.rs
@@ -1,23 +1,20 @@
 pub fn part_one(input: &str) -> Option<u32> {
     let inlines = input.split("\n");
     let mut count: u32 = 0;
-    let mut elves: Vec<u32> = vec![];
+    let mut prev: u32 = 0;
     for l in inlines {
         match l {
-        "" => { elves.push(count); count = 0},
+        "" => {
+            if count > prev {
+                prev = count;
+            }
+            count = 0;
+        },
         _ => { count += l.parse::<u32>().unwrap() }
         }
     }
 
-    let mut largest = 0;
-
-    for e in elves {
-        if e > largest {
-            largest = e;
-        }
-    }
-
-    Some(largest)
+    Some(prev)
 }
 
 pub fn part_two(input: &str) -> Option<u32> {
@@ -58,6 +55,6 @@ mod tests {
     #[test]
     fn test_part_two() {
         let input = advent_of_code::read_file("examples", 1);
-        assert_eq!(part_two(&input), None);
+        assert_eq!(part_two(&input), Some(45000));
     }
 }