loops.py

try:
    n = input()
    n = int(n)

    if n >= 1:
        for i in range(n):
            print(i ** 2)
    else:
        print("Invalid input.")

except ValueError:
    print("Invalid input.")