Update main.hs