Update problem_2.c