修改接口，完善代码