Adding solution for problem 2