Some changes to gen_prefill.py

Did not mean to change README.
Made some changes in gen_prefill.