整理一下程式碼