Create prompt.py