small updates in notebook