Updated Tree Component code.