added solution for problem 2