plt.axis([0, 6, 0, 0.4])