Work for EECS5760 AES in Python