added solution for problem 4