Added entity class