From 9ab83c46ac5a857a2c82c0735fae64a91d19039f Mon Sep 17 00:00:00 2001
From: Pete Shadbolt <hello@peteshadbolt.co.uk>
Date: Thu, 17 Nov 2016 20:50:19 -0800
Subject: [PATCH] Fix up some docstrings

---
 abp/graphstate.py | 22 ++++++++++------------
 1 file changed, 10 insertions(+), 12 deletions(-)

diff --git a/abp/graphstate.py b/abp/graphstate.py
index baa1c9d..0db2d26 100755
--- a/abp/graphstate.py
+++ b/abp/graphstate.py
@@ -62,18 +62,9 @@ class GraphState(object):
         self._del_node(node)
 
     def _add_node(self, node, **kwargs):
-        """ Add a node. By default, nodes are initialized with ``vop=``:math:`I`, i.e. they are in the :math:`|+\\rangle` state.
-
-        :param node: The name of the node, e.g. ``9``, ``start``
-        :type node: Any hashable type
-        :param kwargs: Any extra node attributes
-
-        Example of using node attributes ::
-
-            >>> g.add_node(0, label="fred", position=(1,2,3))
-            >>> g.node[0]["label"]
-            fred
-
+        """ Add a node. 
+        
+        By default, nodes are initialized with ``vop=``:math:`I`, i.e. they are in the :math:`|+\\rangle` state.
         """
         if node in self.node:
             print "Warning: node {} already exists".format(node)
@@ -94,6 +85,13 @@ class GraphState(object):
         :param kwargs: Any extra node attributes
 
         By default, qubits are initialized in the :math:`|0\\rangle` state. Provide the optional ``vop`` argument to set the initial state.
+
+        Example of using node attributes ::
+
+            >>> g._add_node(0, label="fred", position=(1,2,3))
+            >>> g.node[0]["label"]
+            fred 
+
         """
         kwargs["vop"] = clifford.by_name[
             str(kwargs.get("vop", "hadamard"))]  # TODO: ugly