Minor corrections/clarifications to diagram