improving the model