Help me in solving PSPP77 problem

My issue

plsease give sample models

My code

a=int(input())
b=int(input())
c=(a**2+b**3)+(a**2+b**3)**2+(a**2+b**3)**3
print(c)

Learning course: Learn Python
Problem Link: CodeChef: Practical coding for everyone

@chandru_2301
U have to do it like this.

# Solution as follows

for _ in range(3):
    A, B = map(int, input().split())
    C = A**2 + B**3 + (A**2 + B**3)**2 + (A**2 + B**3)**3
    print(C)