add_index function

Add arm indices and agent identifiers to a dataset