Update plot.py

added credit