add solution for problem 002