Work in progress: Finish encrypt in aes.py