added diversity data set test