TODO: Add data description here. Do we need to update data files below to not include test set?