The python notebook needs to be updated

[dyokelson]

Since refactoring so much of the script, the python notebook needs to be updated accordingly to just use the script/library functionality.