[Haskell-beginners] haskell solution for assignment