Add figures for diagrams