Fix State #draw method to return the created node.

This restores a previous behavior which allowed for simple customization
of the generated graph.