Improved tool part localization