Improving Model Classification by Optimizing the Training Dataset