Minor tweaks to notebook for Workshop 2