Added the policy gradient