Simplify Main.hs