結果 : how to write square function in python