Added a header to MainStage.hx and some polishes