[Haskell-beginners] Can we make this program better (and shorter)?