Update prob2.c