Update main.h

edit