Update notebook for clarification