Updated README and notebook.