Updates to session 4 lecture notebook