update readme and add training data