python networkx draw tree