modification in main.c