input: Write a function to generate an RSA public/private key pair and save them to files named 'public.pem' and 'private.pem'. The function should use the Crypto library to generate the key pair, and the keys should be of 2048 bits. The public key should be saved in 'public.pem' and the private key should be saved in 'private.pem'.
  
  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)
  
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)