Added graphing via pylab