changes in main.c

changes in main.c