Update curve to import matplotlib.pyplot rather than matplotlib directly