updated annealing_mpi.py