modify plot.py
modify plot.py