Update main.c

First update main.c