diff --git a/pxl_scripts/px/net_flow_graph/net_flow_graph.pxl b/pxl_scripts/px/net_flow_graph/net_flow_graph.pxl
index 21f6ce4c132..af3117ed021 100644
--- a/pxl_scripts/px/net_flow_graph/net_flow_graph.pxl
+++ b/pxl_scripts/px/net_flow_graph/net_flow_graph.pxl
@@ -25,4 +25,5 @@ def net_flow_graph(start: str, ips: list, grouping_entity: str):
     df = df[df['from_entity'] != '']
     # Look up the names of the remote address.
     df.to_entity = px.nslookup(df.remote_addr)
+    df.bytes_total = df.bytes_sent + df.bytes_recv
     return df.drop(['remote_addr'])
diff --git a/pxl_scripts/px/net_flow_graph/vis.json b/pxl_scripts/px/net_flow_graph/vis.json
index 2d7a65b88f1..3960f8d40e9 100644
--- a/pxl_scripts/px/net_flow_graph/vis.json
+++ b/pxl_scripts/px/net_flow_graph/vis.json
@@ -56,7 +56,18 @@
         "adjacencyList": {
           "fromColumn": "from_entity",
           "toColumn": "to_entity"
-        }
+        },
+        "edgeWeightColumn": "bytes_total",
+        "edgeColorColumn": "bytes_total",
+        "edgeThresholds": {
+          "mediumThreshold": 50000,
+          "highThreshold": 500000
+        },
+        "edgeHoverInfo": [
+          "bytes_total",
+          "bytes_sent",
+          "bytes_recv"
+        ]
       }
     },
     {
@@ -73,4 +84,4 @@
       }
     }
   ]
-}
\ No newline at end of file
+}