first stab at kNN solution with some passing unit tests
another test ... original code just worked :)