# here is a module

def my_f(x):
    return x ** 2
# you can run it from command ilne!

if __name__ == "__main__":
    if my_f(5) == 25:
        print("OK")
    else:
        print("Not OK")

# Output:
# OK
