Working the undo feature