small change to diagram generation