Improve organization of code

[Tom25]

Update the Python scripts to use functions and more organized code constructs. Not super high-priority, but something that will be nice to have for potential future contributors.