Add docstring for add_node function

Add docstring to add node function