add dataset class