Restored main to how it should be