began analysis
generated train and test data
added classification
fixed test data
added tests file
improved classification