Updated the models