Add retrainability test