Update KNNPrediction.java