Simplified Main.hs