Update curve_fitting.py