Remove classes when done