TODO Tests Node Context Node structs functions to create nodes nodes should take in context and state each node should have an id run train init state