minor fixes on the notebook