Rename fig.py to PSet2_fig.py