Added KDTree to compare_velocity_inversion.py