修改bug，整合代码