Update final.ipynb