added solution for the problem 2