def square(nums):
    return [ x*x for x in range(nums) ]

print(square(9))