Improving the model