Proposed changes to flash_algo_gen.py