A little change to the model