Added enums to the generator