Edits to Fig 7 and minor edits elsewhere.

I can issue a PR against your branch if you'd like but it is likely easier/quicker for you to put it in