Update main.c

Add some comments...