modify the main.c