Minor edits to state update section