Added get method for data sets and test