Implementation of AES algorithm in Python