Minor plot edits