update notebook: better docstring