X-Git-Url: https://git.njae.me.uk/?a=blobdiff_plain;f=advent02.hs;h=d4abc555865e2af68e70d161f441f9edd7e6a199;hb=4ed1e3133856105db52d16c8f6adb60b1ebde920;hp=99df10c23390948b8ff3adaf72c3493a3796f041;hpb=f5a48c6a7560298b0eebcd7c47d7edc0506bddb2;p=advent-of-code-16.git

diff --git a/advent02.hs b/advent02.hs
index 99df10c..d4abc55 100644
--- a/advent02.hs
+++ b/advent02.hs
@@ -1,3 +1,5 @@
+module Main(main) where
+
 import Data.Array.IArray
 
 -- Row 1 is top, column 1 is left
@@ -21,7 +23,7 @@ kb2 = ["xxxxxxx",
 enumerate = zip [0..]
 
 mkKeyboard :: [String] -> Keyboard
-mkKeyboard kb = array ((0, 0), (length kb - 1, length kb - 1))
+mkKeyboard kb = array ((0, 0), (length kb - 1, length (kb!!0) - 1))
     [((i, j), c) | (i, r) <- enumerate kb, (j, c) <- enumerate r]
 
 keyboard1 = mkKeyboard kb1