Python problem 3

With a given integral number n, write a program to generate a dictionary that contains (i, i*i) such that is an integral number between 1 and n (both included). and then the program should print the dictionary.

class Solution:
    def square1(self,n):
        dict1={}
        for i in range(1,n+1):
            dict1[i]=i**2
        return(dict1)


if __name__=="__main__":
    kk=Solution()
    print(kk.square1(5))