Updated main.h