cc @kimbochen The refactor in #145 introduced a singular Python script with lots of logic. We can try to simplify it as a p2 task.