working code and data set