Do problem 3 in Haskell.