updated Main.hs