notebook updated with the code