added categories to the predictions