minor updates

minor updates to notebook cood