Add more test cases for arrays