Trouble Implementing AES Key Expansion in Python

Question

I am trying to implement the AES Key Expansion in python, but I am having some trouble. Here is the code I\'m using:

    def print_matrix(m):
         #Just t