@@ -123,16 +123,12 @@ prepareOutput s =
123123
124124prepareCode ∷ String -> String
125125prepareCode code =
126− if startsWith "module" code then
127− code
128− else
129− if startsWith "import " code then
130− moduleMain <> code
131− else
132− if (startsWith "main =" || contains (Pattern "\nmain =")) code then
133− moduleMain <> importBasics <> code
134− else
135− moduleMain <> importBasics <> spyMain <> code
126+ let prepare
127+ | startsWith "module" code = code
128+ | startsWith "import " code = moduleMain <> code
129+ | (startsWith "main =" || contains (Pattern "\nmain =")) code = moduleMain <> importBasics <> code
130+ | otherwise = moduleMain <> importBasics <> spyMain <> code
131+ in prepare
136132 where
137133 moduleMain = "module Main where\n"
138134