Updated diagram.ipynb