Update main.c

correction