Update main.h

A few small changes