改進函數代碼