Update main.c

add some stuff