Extend undo stack support