Adding entity class