Python 3 fix plot.py