結果 : python code for square of a number