Add just pylab mock back