After messing around with custom mobs, I decided to make a bee which belongs to a team. I created this bee using this command: /give @p bee_spawn_egg{EntityTag:{id:"minecraft:bee",Team:Red}}. I also created Red Team which the bee would belong to; additionally, I gave Red Team a prefix, Red. I spawned in the Bee using the spawn egg, and the prefix was present. I eventually decided to place down a hive, and the bee after pollinating a flower flew right inside. I then provoked the bee to leave the hive by removing it in survival, and the bee had lost its prefix; this could also mean that the bee lost its team after leaving the hive.