Adding back classes