[Haskell-beginners] Converting an imperative program to haskell