input: Write a function to generate an RSA public/private key pair and save them as files named 'public.pem' and 'private.pem'.
  
  from Crypto.PublicKey import RSA
  
  def generate_key_pair():
   key = RSA.generate(2048)
   pub_key = key.publickey().export_key().decode()
   with open('public.pem', 'w') as file:
    file.write(pub_key)
   
   priv_key = key.export_key().decode()
   with open('private.pem', 'w') as file:
    file.write(priv_key)
  
  generate_key_pair()
output: def generate_key_pair():
   key = RSA.generate(2048)
   pub_key = key.publickey().export_key().decode()
   with open('public.pem', 'w') as file:
    file.write(pub_key)
   
   priv_key = key.export_key().decode()
   with open('private.pem', 'w') as file:
    file.write(priv_key)