added solution for problem 5