Update main.c

Change inclusion of header file to main.h