Cleaned up code in kd_tree_scene.py a bit