Correcting main class