def square(num):
    return num * num