Составить программу которая выводит на экран квадрат данного числа если оно оканчивается на цифру 3

print ("Введите число " + "\n")

a = int (input ("> "))

a_1 = a  

ml = list  (str(a))

element = ml [len (ml) - 1 ]

i = int (element)

if i == 3 :  

   a *= a

   print  ("Квадрат числа " + str (a_1) + " равен : " + str (a))

else :

   print ("Число не оканчивается на цифру 3")