Update generator.sh