Update problem_4.c