Update assignment2/main.c