main class updated