Better support for NumPy indexes