Improve model performance