Retraining with our own images