Finished moving all the MPI to modules. Implemented the scatter/gather