Adding category predictor