improve diagram_viewer.py