Updating visitor, fixing issue with undo