Tree rendering with python