add solution for problem-2