Add remove_sub_data function