Add working AES key expansion in AES-encrypt.py.