add the function used to load data set