重構程式碼
改成類別