Added comments, played with Test.ipynb