Tree Function In Python