Added plt.figure() to keep aspect.