Updated the entity module