Modularize code

[lowderchris]

Convert the code from the work-in-progress script into a more focused and modularized code. In particular, some blocks of code that are currently mirrored for different datasets can be simplified.