Improve data set