Adding elastix-based distortion correction