Changed main class