Coded the solution to problem 2