Edits and figure updates

Some edits from Colleen and changes to make figures more clear.