Update main.hs

more minor fixes