Generator upgrade