update main.hs