Help me in solving PYTH32 problem

My issue

Update your code below this line

s = 14
area = s*s
cost = area//7
print(area)
print(cost)

My code

# Update your code below this line
s = 14
area = s*s
cost = area//7
print(area)
print(cost)

Learning course: Learn Python Programming
Problem Link: https://www.codechef.com/learn/course/rcpit-python/RCPITPY05/problems/PYTH32

I enhance it for better readability and flexibility

# Update your code below this line
s = 14
area = s * s  # Calculating the area of a square
cost_per_unit = 7  # Define cost divisor as a variable
cost = area // cost_per_unit  # Calculate cost

print("Area of square:", area)
print("Total cost:", cost)