Update main.c file