Square Pattern Code In Python