Add type signatures in Main.hs