Value iteration seems to work better, however does not converge and is very expensive