python area of a square