Add key generation and decryption