Continued setting up undo/redo framework