Update tool.py

add yaml header parse