small fixes to notebook