First commit: python script and notebook.
First commit on github.
Updated the readme, and some fine tuning of the script.