Undo/redo now working