Add numpy indexing support to __getitem__