diff --git a/plot_single_treansmission_paper.py b/plot_single_treansmission_paper.py
index e66f649..22311cb 100644
--- a/plot_single_treansmission_paper.py
+++ b/plot_single_treansmission_paper.py
@@ -235,7 +235,7 @@ if __name__ == "__main__":
         ax_stacked = transmission_df[bw_cols].plot.area(stacked=True, linewidth=0, ax=ax00)
         ax00.set_ylabel("bandwidth [MHz]")
         #ax.set_xlabel("time [minutes]")
-        ax00.set_xlim([0, transmission_df.index[-1]])
+        #ax00.set_xlim([0, transmission_df.index[-1]])
         ax00.xaxis.grid(False)
 
 
@@ -266,9 +266,14 @@ if __name__ == "__main__":
             ax2.legend(lns_ax0, labs_ax0, ncols=9, fontsize=9, loc="upper right", frameon=legend_frame)
             #ax0.set_zorder(100)
 
-            lns_ax00 = [ax_stacked]
-            labs_ax00 = ["5G bandwidth", "4G bandwidth"]
-            ax00.legend(lns_ax00, labs_ax00, ncols=3, fontsize=9, loc="upper center", frameon=legend_frame)
+            #lns_ax00 = [ax_stacked]
+            #labs_ax00 = ["5G bandwidth", "4G bandwidth"]
+            #ax00.legend(lns_ax00, labs_ax00, ncols=3, fontsize=9, loc="upper center", frameon=legend_frame)
+
+            L = ax00.legend(ncols=2, fontsize=12, frameon=False)
+            L.get_texts()[0].set_text("5G bandwidth")
+            L.get_texts()[1].set_text("4G bandwidth")
+
             #ax00.set_zorder(100)
             plt.savefig("{}{}_plot.eps".format(args.save, csv.replace(".csv", "")), bbox_inches="tight")
         else: