improve the shiftRows figure