Add suport for node tracking