First pass at Assignment #5
Correcting neighbor search for KdTree.