update the output to have the classification