Update main.c

added some comments