adding in diagrams