Ones again, working policy iteration version