修改bug以及精简代码

修改bug以及精简代码