Add plots.py