Initial commit, everything works, just need to review training data