added support for editing classes