Adding more test cases
Allow for retraining when data is modified