Minor edits to updating code