Added a python notebook for fitting