Undo and redo will work well now!