ENH: Update notebook
ENH: Update notebook