Improvement the model