Updates to notebook for tutorial