Adds initial work on 2D halo exchange example