Updated problem.c