Cleaned up Main.hs with do notation