started improving the mpi version