結果 : how to write square and cube in python