Add function for loading sampled set