Tidying

[advent-of-code-16.git] / advent02.hs

diff --git a/advent02.hs b/advent02.hs
index 99df10c23390948b8ff3adaf72c3493a3796f041..d4abc555865e2af68e70d161f441f9edd7e6a199 100644 (file)
--- 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