add diagramDrawer class