Added in digraph support and node sizing