New policy iteration algorithm
Minor improvement