[Haskell-beginners] Better way to write this?