small comment and clarity edits