Small code review for  main.hs