modify main class.