Added test entity class