change argument to `index` in subset method