Added editing of classes