minor tweaks to notebook