Add key generation testing option