Added new classes undo/redo functionality