Adding solution for problem 6