added a header to main.c