add a diagram, and some edits