Actually fixed main class issue