Fixed AddDiagram not creating model