updated entity class