Adding test on dataset