Дано число п. Из чисел 1, 4, 9, 16, 25, ... напечатать те, которые не превышают n. Python

Ниже представлен код на Python, который выводит числа 1, 4, 9, 16, 25, и т.д., которые не превышают заданное число n:

import math

def print_squares(n):
    i = 1
    while i**2 <= n:
        print(i**2)
        i += 1

n = int(input("Введите число n: "))
print_squares(n)

Просто запустите программу и введите число n, и она выведет все квадраты чисел, которые не превышают n.