Add graph and node class