Added header to main.c
did some cleaning up.