Made some changes to new class selection menu