Working on getting Add Class button to work again correctly