add solution for problem 3