how to write flow of execution in python examples

Example: how to write flow of execution in python

def pow(b, p):
    y = b ** p
    return y

def square(x):
    a = pow(x, 2)
    return a

n = 5
result = square(n)
print(result)