add Diagram class