Add main.h to main.c