added test to entity class