修改代码以及readme

修改代码以及readme