Basic MPI works, but without load-balancing (commented out for now)