Fully running classification.
Needs some score boosting,
Train accuracy = 0.9921787709497206
Test  accuracy = 0.6104910714285714