Added some info in the README and edited notebook to reflect correct link to the github repository.