Added data set and loader script