Add test for array indexing