@WIP refactor and get_data_sets method