cost-based-ml/cost_based_ml.py (2 lines): - line 26: # TODO add error checking, assuming success - line 47: # TODO a bit hacky, find a better way to parse