Add entity classes