change Model to Entity