diff --git a/plot_transmission_timeline.py b/plot_transmission_timeline.py
index dad9817..7935ea0 100755
--- a/plot_transmission_timeline.py
+++ b/plot_transmission_timeline.py
@@ -148,9 +148,7 @@ if __name__ == "__main__":
     print(transmission_df)
 
     # key for columns and level for index
-    goodput_df = transmission_df["payload_size"].groupby(pd.Grouper(level="datetime", freq="{}s".format(args.interval))).sum().dropna()
-    print(goodput_df["payload_size"])
-    exit()
+    transmission_df["goodput"] = transmission_df["payload_size"].groupby(pd.Grouper(level="datetime", freq="{}s".format(args.interval))).transform("sum")
 
 
     #transmission_df["goodput"] = transmission_df["payload_size"].rolling("{}s".format(args.interval)).sum()