Tomáš Kuzma

class Rectangle:

    def __init__(self, a, b):
        self.a = a
        self.b = b

    def area(self):
        return self.a*self.b


class Square(Rectangle):

    def __init__(self, side):
        # self.a = side
        # self.b = side
        super().__init__(side, side) # better


A = Rectangle(4, 5)
B = Square(6)

print(A.area())
print(B.area())