add diagram1.py