py codes

ipynb!!