Add test for array