modify main class