Update main.h