added test for dataset