Updates for new changes to pred_setA1.py