Nonlinear Fit Python Code