modify AES.py, basic architecture