Add 2D test case