added missing curve_fit import