Fix problem_4