Remove unneeded line from plot.py