update main.c

main.c