Tweaks to session 4