Updated the models gem