1 files changed, 6 insertions, 4 deletions
diff --git a/src/2016/day2/aoc.cpp b/src/2016/day2/aoc.cpp
index 3b5b28b..a73f26f 100644
--- a/src/2016/day2/aoc.cpp
+++ b/src/2016/day2/aoc.cpp
@@ -2,14 +2,16 @@
 
 namespace aoc2016 {
 
-int day2(line_view file) {
+int day2(line_view file, char ds[]) {
   int d{0};
-  int l{5};
+  size_t i{0};
+  char l{5};
   keybad pad;
-  per_line(file, [&d, &l, &pad](line_view lv) {
-    l = pad.touch(l, lv.line);
+  per_line(file, [&d, &l, &pad, &i, &ds](line_view lv) {
+    l = pad.touch(l, lv.line, pad.digit_keys);
     // printf("%d\n", l);
     d = d * 10 + l;
+    pad.touch(i++, ds, lv.line, pad.diamond_keys);
     return true;
   });
   return d;