Add encryption functionality