Update main.hs

Changes..