Updated test_fit.ipynb