Fixed problem 4 solution.