reverted back to KNN, added extra features