minor updates in notebook