slight updates to notebook