UMass's code used in the KBA task