Update assn4.c