some updates in the notebook