Add plt.figure() call to initializer.