Update linked-lists-python.py

Author: pedropregueiro

linked-lists-python/linked-lists-python.py

@@ -49,11 +49,10 @@ def add_after(self, target_node_data, new_node):
             raise Exception("List is empty")
 
         for node in self:
-            if node.data != target_node_data:
-                continue
-            new_node.next = node.next
-            node.next = new_node
-            return
+            if node.data == target_node_data:    
+                new_node.next = node.next
+                node.next = new_node
+                return
 
         raise Exception("Node with data '%s' not found" % target_node_data)