Update main.c

Some changes