Add split_data function