BUG: Fixed the bellman_parallel.py by added uncommented code for all Niter.

Also moved return line from bellman_parallel(2).py.main() to be run by all processes, not just rank==0. This is necessary because if all processes don't exit the function, the program doesn't continue to run.