added class back