Revert "test"

This reverts commit fc0b06827b4ee36658c159a2d58beb9843020f57.
Add entity classes