Add AES encryption functions to encrypt.py

* And tests.