changes in plot.py