finish the data_set class. add tests.