Updated generator shell script