Inject the entity class and add docblocks