Small update to notebook