added training data and script