plot.py now PEP8