Improve the notebook