Add in diagrams