Update k-nn.py