Improved undo action for ST2