Add feature to generate key pairs