Corrected the main class.