Just copy your current generator, and make it unactive on game start and make it so it activates when receiving the upgrade. i think that should work.
Just copy your current generator, and make it unactive on game start and make it so it activates when receiving the upgrade. i think that should work.