fix hw3/4.c