Minor changes and including notebook in the README