Update sample_tools.py