Move plotting functions in graph.py into new plot.py file so that the former doesn't need to import matplotlib.