結果 : how to write main function python