changed the main class