some modifications in the model