Add diagrams and change wording