Update main.c

My changes good yes?