updated version of session 4 notebook