I added the new entity classes