Update prompt.py