updated notebook
updated readme