Adding feature indices to data.py