Fix main claass