Improve the policy model download