sure, this looks simplification looks fine. What if you do `%pylab inline` ? Does that solve the issue?