Python draw tree graph