Updating the sample notebook after merging Scott's PR.