working towards finishing KNN regression