Added an entity class