Added problem 5 solution