BUG: Remove extra unneccasary "pylab.figure()"

It created an extra window in IPython notebook.