Completed problem 3 in Haskell