Working on the migration of the ImageClassification routine.