Modify Main.hs