policy improvement almost working