Added working draft of plot.py