minor changes in the model