Updated figure3.ipynb