Added a solution for problem 5