@@ -123,16 +123,12 @@ prepareOutput s =
124prepareCode ∷ String -> String
124prepareCode ∷ String -> String
125prepareCode code =
125prepareCode code =
126− if startsWith "module" code then
127+ | startsWith "module" code = code
129− if startsWith "import " code then
128+ | startsWith "import " code = moduleMain <> code
132− if (startsWith "main =" || contains (Pattern "\nmain =")) code then
129+ | (startsWith "main =" || contains (Pattern "\nmain =")) code = moduleMain <> importBasics <> code
133− moduleMain <> importBasics <> code
135− moduleMain <> importBasics <> spyMain <> code
130+ | otherwise = moduleMain <> importBasics <> spyMain <> code
137 moduleMain = "module Main where\n"
133 moduleMain = "module Main where\n"