Add collapse_vec function