Model retraining

shape