polishing a few more things of the bellman iteration
Still some issues, but the first iteration is working and takes 50 seconds