Update complex_fit.py