修复

修复上个提交的bug，以及一些修改