change requirements
add training data
nn stuff
trainer