Added a solution for problem 6