Add problem 2 solution.