Update gen_output.sh