Add ImageTrainer class