Added test_fit.py