結果 : how to code a square in python