improved training set and scripts