Add class editing features