Update fit_transits.py