Able to find k nearest neighbors
Fix getting k nearest