changed update halo kernels to be similar to 3d version