Updated models
Corrected KNN