Update prod.py