Update main.c

Final update.