Update Main.hs