Moved some variables to state and tried to implemented custom node and link rendering, but doesn't get called for some reason