結果 : python how to write main function