Add automation for custom nodes