Updated figure4.ipynb