Fixed bug in Main.hs