Add a couple flows for new node type