Updates to notebook