Correcting main class name