diff --git a/plot_cluster_life.py b/plot_cluster_life.py
index fb87f10..175713f 100644
--- a/plot_cluster_life.py
+++ b/plot_cluster_life.py
@@ -14,7 +14,10 @@ if __name__ == "__main__":
         for k in sorted(map(int, raw.keys())):
             idx.append(k)
             data.append(np.array(raw[str(k)]))
-    fig = plt.figure(figsize=(4.5,3))
+    if cb == 1:
+        fig = plt.figure(figsize=(7.5,3))
+    else:
+        fig = plt.figure(figsize=(3,3))
     ax = fig.gca()
     bp = ax.boxplot(data, labels=idx)#, boxprops=dict(facecolor='lightblue'))
     ax.plot(np.arange(1, len(idx)+1), [np.mean(r) for r in data], '^-', color='green', linewidth=1, alpha=0.8)