Added undo/redo mock. ref #337