Fixed a minor bug in the notebook and reset the values