add solution for problem 2