Try to make this work for unfinished diagrams (assume it is not finished / could have chunks & loose / missing pieces) This will involve a button that can "clean up" the look of the diagram so it looks more organized.