modified: fig2.py