k_p1 = np.mod(k_0 + 1, 3)