[Haskell-beginners] Help me improve my code