add_index function

Add follow-up time and arm indices to a dataset