Added plot.py and other necessary files