add dataset for the variational program