Update fit.py