Try this with pylab?