modify the main.c in step3