Update main.c

add comment and puts