Погите решить задачу по информатике

a = 7

if a >= 2 and a <= 17:
b = 3
p = a * a + b * b
else:
b = 5

p = (a + b) * (a + b)
print(p)