Optimize undo session implementation