結果 : how to make a square function in python