Agreed it could use improvement, I'm working on fine-tuning the model to create better diagrams (this is using the latest 3.5 turbo model which typically requires a couple iterations before it gets to a valid diagram).