Pass BinaryTree instead of KDTree / BallTree