[Haskell-beginners] Help with improving a program