draw a square py