Edits for reviewer 2

Edits for reviewer 2