Adding .py version of ipynb