Improve docs for array indexing primops