update main.c
bug fixed