Python now allows pylab argument