Update main.c

couble errors fixed