1 files changed, 11 insertions, 6 deletions

diff --git a/2021/day9/src/main.rs b/2021/day9/src/main.rs
index 5d84be9..524fb1b 100644
--- a/2021/day9/src/main.rs
+++ b/2021/day9/src/main.rs
@@ -43,14 +43,19 @@ fn main() {
                 .get(ndx)
                 .unwrap_or(&9);
 
-            let down = match points.get(idx + 1) {
-                Some(line) => line.get(ndx).unwrap_or(&9),
-                None => &9,
-            };
+            let down = points
+                .get(idx + 1)
+                .unwrap_or_else(|| &nine)
+                .get(ndx)
+                .unwrap_or_else(|| &9);
 
-            let left = height.get((ndx as isize - 1) as usize).unwrap_or(&9);
+            let left = height
+                .get((ndx as isize - 1) as usize)
+                .unwrap_or_else(|| &9);
 
-            let right = height.get(ndx + 1).unwrap_or(&9);
+            let right = height
+                .get((ndx as isize + 1) as usize)
+                .unwrap_or_else(|| &9);
 
             if num < up && num < down && num < right && num < left {
                 risk_level += 1 + *num;