Added problem 4 solution.