Small updates to the models