[Haskell-beginners] shortening code