This script doesn't work on a Jupyter notebook, how do I modify it to work?