Complex.

class Complex:
   def __init__(self,real,im):
      self.real = real
      self.im = im
 
   def print(self):
      i = self.im
      if i > 0:
         print(self.real,end="+")
      else:
         print(self.real,end=" ")
      print(self.im,end="i")
      print()
       
   def add(self,b):
      r = self.real + b.real
      ic = self.im + b.im
      add = Complex(r,ic)
      return add
 

def setup():
   a = Complex(5,3)
   b = Complex(2,-1)
   c = a.add(b)

   a.print()
   b.print()
   c.print()

   assert c.real == 7 and c.im == 2
setup()