trainmodel and test model improvement