How To Write A Square In Python