problem-4 solution is added