Add key generation methods