Update main.hs

simplified