Create plot.py