add entity class parameter