Update main.c

error fix