Updated main.c