changed main class