Update to work with the new NodeCG API