correct main class