Adding the Python script and improving the notebook.
Adding the Python script and improving the notebook.