Adding solution for problem 3