diff --git a/day01/sol.ivy b/day01/sol.ivy
index 40aca65..b13957d 100644
--- a/day01/sol.ivy
+++ b/day01/sol.ivy
@@ -4,5 +4,5 @@
 max/ +/ input
 
 # part 2
-sums = +/ input
-+/ sums[3 take down sums]
+op n top v = v[n take down v]
++/ 3 top +/input