add solution for problem 4