Modify main.c