Update problem2.c