Update UKF Tractography

Fix testing