Update main.c

Last changes...