Edits to Fig 7 and minor edits elsewhere.

Only thing I noticed when looking through the PR