Edit generator.sh