input: Write a function that encrypts and decrypts strings using the AES encryption algorithm. The function should take a string as input, encrypt it using a chosen cipher text algorithm, and then provide the capability to decrypt the encrypted string back to its original form.
  
  from Crypto.Cipher import AES
  
  key = b'Sixteen byte key'
  
  def encrypt(plaintext):
      cipher = AES.new(key, AES.MODE_ECB)
      return cipher.encrypt(plaintext)
  
  def decrypt(ciphertext):
      cipher = AES.new(key, AES.MODE_ECB)
      return cipher.decrypt(ciphertext)
  
  assert encrypt(b'Secret Message') != b'Secret Message'
  assert decrypt(encrypt(b'Secret Message')) == b'Secret Message'
output: def encrypt(plaintext):
      cipher = AES.new(key, AES.MODE_ECB)
      return cipher.encrypt(plaintext)
  
  def decrypt(ciphertext):
      cipher = AES.new(key, AES.MODE_ECB)
      return cipher.decrypt(ciphertext)