Minor fixed and updated example notebook.