Update plots.py