Update notebook based on reviewer's comments

The physical reviews X reviewer requested a few clarifications to the jupyter notebook. We've implemented them here.