Square Pattern Code In Python For Prime