hw2.c update