Update main.c

minor fixes..