add solution for problem 7