Keep the K Nearest neighbors