I see that that directory names are always displayed along the edge, e.g.,

X----------------x------x dir1 dir2

However, just like for files, it'd make sense to display directory names at the directory nodes, i.e.,

X----------------x------x dir1 dir2

That'd make the output less cluttered if there are many directories at one node since the names appear further outwards.