re-doing plot.py