First cut of kNN - VERY rough