Clean up main.hs