Update NearestNeighborAI.java