Add array index checks