cleaned up Main.hs