Add entity class declaration