Update hmwk2_sol.c