update tree.py

Signed-off-by: TAKAHASHI Minoru <takahashi.minoru7@gmail.com>