Update main.c

fix bugs