diff --git a/src/main.rs b/src/main.rs
index 1c02018..5001ed8 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -97,11 +97,7 @@ fn App() -> Element {
                             td {
                                 class: "number",
                                 b {
-                                    if row.len() <= 1 {
-                                        {row.iter().sum::<u16>().to_string()}
-                                    } else {
-                                        {row.iter().take_while(|&&n| n != 5).sum::<u16>().to_string()}
-                                    }
+                                    {row.iter().enumerate().take_while(|(c, n)| *c < 1 && **n != 5).map(|(_, n)| n).sum::<u16>().to_string()}
                                 }
                             }
                         }