diff --git a/src/main/java/Simulator.java b/src/main/java/Simulator.java
index 1f6fdd0..b3e37d1 100644
--- a/src/main/java/Simulator.java
+++ b/src/main/java/Simulator.java
@@ -149,22 +149,37 @@ public class Simulator implements GraphModificationFunctions {
         betaVertices.forEach(graph::addVertex);
         //add edges
         for(Vertex a: alphaVertices) {
-            for(Integer well: a.getRecord().getWells()) {
-                for (Vertex b: betaVertices) {
+            for(Vertex b: betaVertices) {
+                double weight = 0.0;
+                for (Integer well: a.getRecord().getWells()) {
                     if (b.getRecord().isInWell(well)) {
-                        DefaultWeightedEdge edge = graph.getEdge(a, b);
-                        if (edge == null) {
-                            edge = graph.addEdge(a, b);
-                            graph.setEdgeWeight(edge, 1.0);
-                        }
-                        else {
-                            double weight = graph.getEdgeWeight(edge);
-                            graph.setEdgeWeight(edge, weight + 1.0);
-                        }
+                        weight += 1.0;
                     }
                 }
+                if (weight != 0.0) {
+                    DefaultWeightedEdge edge = graph.addEdge(a, b);
+                    graph.setEdgeWeight(edge, weight);
+                }
             }
         }
+//        //add edges
+//        for(Vertex a: alphaVertices) {
+//            for(Integer well: a.getRecord().getWells()) {
+//                for (Vertex b: betaVertices) {
+//                    if (b.getRecord().isInWell(well)) {
+//                        DefaultWeightedEdge edge = graph.getEdge(a, b);
+//                        if (edge == null) {
+//                            edge = graph.addEdge(a, b);
+//                            graph.setEdgeWeight(edge, 1.0);
+//                        }
+//                        else {
+//                            double weight = graph.getEdgeWeight(edge);
+//                            graph.setEdgeWeight(edge, weight + 1.0);
+//                        }
+//                    }
+//                }
+//            }
+//        }
         if(verbose){System.out.println("Graph created");}
         //stop timing
         Instant stop = Instant.now();