Update main.c

Comments edit