Напишите программу, которая получается два целых числа a и b и выводит квадраты всех натуральных чисел в интервале от a до b. Python

a = int(input())

b = int(input())

for i in range(a, b + 1):

   print(i, ':', i ** 2)

Программа:
Python:

a = int(input('a = '))

b = int(input('b = '))

for i in range (a, b + 1):

   print(i * i)