to make this easier u can print (i**2)
#where it prints i^2
OR try printing (i * i) instead of (n*n)
hope this helps!