clean up plots.py a bit

Issues found looking at #177 none of which seem to actually fix that problem.