updating tests for new data set