Update main.c

updated to his main