Working on output of model predictions.