updates based on new generator