code/demo_prompt_args.py