Proposed change to flash_algo_gen.py