/// in wpnspellhit.cpp
   /// from line 411 (void ConsumeHit() function)
   void ConsumeHit(CUnit* attacker, CUnit* target) {

       if(
           target != NULL &&
           !(target->status & UnitStatus::Invincible) &&
           getSpellStatString(target,TechId::Consume,attacker->playerId) == 0 //returning a value != 0 means there was an error
       )
       {

           incrementUnitDeathScores(target,attacker->playerId);
           target->remove();

           if(!(target->status & UnitStatus::IsHallucination)) {

               /// KYSXD
               /// For use with WeaponId namespace
               s16 healtRegen = weapons_dat::DamageAmount[117] * 256; /// WeaponId = 117
               attacker->setHp(attacker->hitPoints + healtRegen);

               u16 energyMax = attacker->getMaxEnergy();

               if((energyMax - attacker->energy) > 12800) //12800 is 50 energy ingame
                   attacker->energy = energyMax;
               else
                   attacker->energy += 12800;

           }

       }

   } //void ConsumeHit(CUnit* attacker, CUnit* target)

   /// In update_unit_state.cpp
   /// from line 83 (else statement of 'if(unit->energy == maxEnergy)')
   else {

       /// KYSXD:
       /// Normal energy regen
       u16 energyRegen = 8;

       /// KYSXD:
       /// For use with UpgradeId namespace
       if(unit->id == UnitId::Hero_InfestedDuran)
       {
           switch(scbw::getUpgradeLevel(unit->playerId, 18)) /// UpgradeId 18
           {
               case 1:
                   energyRegen *= 1.5;
                   break;
               case 2:
                   energyRegen *= 2;
                   break;
               default:
                   break;
           }                  
       }

       /// KYSXD:
       /// New modified regen
       unit->energy += energyRegen;

       if(unit->energy > maxEnergy)
           unit->energy = maxEnergy;

   }