Classification is now working with about 75% accuracy on test data...