ООП, нужно отредактировать код

import math
 
 
class Complex:
    def __init__(self, a, b):
        self.a = a
        self.b = b
 
    def __str__(self):
        return f'Re(z) = {self.a}; Im(z) = {self.b}'
 
    def module(self):
        mod = math.sqrt(self.a ** 2 + self.b ** 2)
        return mod
 
    def arg(self):
        if self.a < 0:
            if self.b > 0:
                z = math.ceil(math.degrees(math.pi + math.atan(self.b / self.a)))
            else:
                z = - math.ceil(math.degrees(math.pi - math.atan(self.b / self.a)))
        elif self.a == 0:
            if self.b > 0:
                z = 90
            elif self.b < 0:
                z = 270
            else:
                z = None
        else:
            z = math.ceil(math.degrees(math.atan(self.b / self.a)))
        return z
 
    def __truediv__(self, other):
        n1 = complex(self.a, self.b)
        n2 = complex(other.a, other.b)
        return n1 / n2
 
 
class Trigonometric(Complex):
    def __init__(self, a, b):
        Complex.__init__(self, a, b)
        self.module = Complex.module(self)
        self.arg = Complex.arg(self)
        self.z = complex(self.a, self.b)
 
    def __str__(self):
        return f"{self.z} = {self.module} (cos({self.arg}) + sin({self.arg}))"
 
 
z1 = Complex(-1, 1)
z2 = Complex(1, -1)
print(z1)
print("|z| =", z1.module())
print('arg z = ', z1.arg(), "°", sep="")
print("z1 / z2 =", z1 / z2)
t = Trigonometric(-1, -1)
print(t)