Update main.c

modify compile error.