Changed main class back