add back classes