fixing small bugs and testing partial run code
Updated ipynb with more data