modify plot.py