def square(x):
    '''
    Returns the square of a number.

    Args:
        x: input number

    Return:
        The square of x
    '''
    
    # block in the function was not indented
    # also, square should be x*x (it was not necessary to fix this though)
    return x*x

print(square(4))
