Trying to integrate the new key generation functionality.