add problem 2 solution