add diagram2.py