Update main.c

modified