added script and data set, updated readme