Add type signature for policyIterationN