Fixed the main class.