Update main.c

4