Added an EnumTrait instead of the BaseEnumaration class