small changes to the model