curve_fit() works