Change predictLabels()