first
training data
changes to main program