[Haskell-beginners] Return function