Added plotting routines in plot.py