Update main.c

Final code