add problem 2 soln