modify main.c