Working on problem 2 Haskell solution.