modify the main class