changed model to entity