Hi,
From having a rough look into the code I figured you did not style the node boxes by css. It would be much easier to change the look to say elliptic nodes.
Please extend the css-ability and also consider adding labels to the connection lines.
Thanks for this great piece of work!