add diagrams code