Added problem 2 solution