Update main.c

fixed things and stuff.