change outputs to be class lables