Update policy.py