Update main.c

edit main.c