Leo: I think shift rows is shifting the rows and not the columns - could be wrong though. Take a look at shift_rows in aes.py - I think this is the desired outcome