結果 : how to write square in python 3