changed the main class.