Adding more functionality to the AES model.
Adding state registers that follows the AES spec with 16 separate bytes.
Adding 16 sboxes with wires.