Small rewrites of notebook from JB's suggestions