update policy.py