add colors for prediction