Implemented adding and removing the specified editing class during editing. All tests pass now.