added code to get training sets
added training sets
added training sets
added training sets
completed training set code