Updated encrypt.py with comments