Full registration pipeline now working in 2D and 3D