Update the notebook to reflect modification in the way we deal with test cases.