Add the entity class and persistence unit