Extending Kilosort for 3D Neural Probes