Update fit_estimator.py