Basics for value_iteration are there, the formula doesn't quite work yet..