Update main.c

changes...