modified the notebook in accordance with the merge of the two scripts