Update curve_fit.py