working on getting initial training data