Create gen_prefill.py