Update main.c

added stuff to main