Update for KdTree
Finishing assignment 5