add logic for all nodes/edges
add api remove functionality