plt.xlabel('step*replica')
replica=range(int(sys.argv[2]))
-colors = cm.rainbow(np.linspace(0, 1, len(replica)))
+#colors = cm.rainbow(np.linspace(0, 1, len(replica)))
+cmap = plt.get_cmap('hot')
+colors = cmap(np.linspace(0, 1, len(replica)*1.4))
for i,c in zip(replica,colors):
yt=x0[r0==i]
xt=(s0+r0*max(s0))[r0==i]