change gen_ans.py to argv format