started migratin code from .py to .ipynb