AES 128-bit implementation in Python