Update fitting.py