Restore plot.py to its former glory

ಠ_ಠ