adding class back in