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