added the solution to problem 2