Update main.c

can run and the result is right!