updated notebook some more
updated notebook some more