Made Jason's changes to the notebook