Simplifies 3D Halo exchange using the fact that a(:,:,i) is a 2D array