added training set creation function