change to main class