Updating the prediction set interface