Add shiftRows function.

Could be cleaner but what the hell. Optimize later.