Update main.c